package com.bihu.bihudemo.controller;


import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.api.ApiController;
import com.baomidou.mybatisplus.extension.api.R;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.bihu.bihudemo.dao.FriendDao;
import com.bihu.bihudemo.entity.Friend;
import com.bihu.bihudemo.entity.User;
import com.bihu.bihudemo.exception.ErrorEnum;
import com.bihu.bihudemo.result.Result;
import com.bihu.bihudemo.service.FriendService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.io.Serializable;
import java.util.List;

/**
 * (Friend)表控制层
 *
 * @author makejava
 * @since 2020-07-20 15:19:29
 */
@RestController
@RequestMapping("friend")
@Api(tags = "好友相关操作")
public class FriendController extends ApiController {
    /**
     * 服务对象
     */
    @Resource
    private FriendService friendService;

    @Resource
    FriendDao friendDao;
    /**
     * 分页查询所有数据
     *
     * @param page   分页对象
     * @param friend 查询实体
     * @return 所有数据
     */
    @GetMapping
    public R selectAll(Page<Friend> page, Friend friend) {
        return success(this.friendService.page(page, new QueryWrapper<>(friend)));
    }


    /**
     * 添加好友的操作
     * @param friend
     * @return
     */
    @ApiOperation("添加好友")
    @PostMapping("/addfriend")
    public Result addFriend(@RequestBody Friend friend){
        int uid=friend.getUserId();
        int fid=friend.getFriendId();
        List<Integer> frinedID = friendDao.getFrinedID(uid);
        if(frinedID.contains(fid)){
            //重复添加的错误
            return Result.error(ErrorEnum.DUPLICATE_KEY);
        }else {
            friendService.save(friend);
            return Result.ok();
        }
    }

    /**
     * 获取用户好友列表
     * @param uid
     * @return
     */
    @ApiOperation("获取好友列表")
    @GetMapping("/listFriend")
    public Result getFriends( @RequestParam("uid")Integer uid){
        List<User> friends = friendService.getFriends(uid);
        return Result.ok().put("friendList",friends);
    }
    /**
     * 通过主键查询单条数据
     *
     * @param id 主键
     * @return 单条数据
     */
    @GetMapping("{id}")
    public R selectOne(@PathVariable Serializable id) {
        return success(this.friendService.getById(id));
    }

    /**
     * 新增数据
     *
     * @param friend 实体对象
     * @return 新增结果
     */
    @PostMapping
    public R insert(@RequestBody Friend friend) {
        return success(this.friendService.save(friend));
    }

    /**
     * 修改数据
     *
     * @param friend 实体对象
     * @return 修改结果
     */
    @PutMapping
    public R update(@RequestBody Friend friend) {
        return success(this.friendService.updateById(friend));
    }

    /**
     * 删除数据
     *
     * @param idList 主键结合
     * @return 删除结果
     */
    @DeleteMapping
    public R delete(@RequestParam("idList") List<Long> idList) {
        return success(this.friendService.removeByIds(idList));
    }
}